import streamlit as st
import os
import sys

# 添加当前目录到系统路径
sys.path.append(os.path.abspath("."))

from app.pages.home import show_home
from app.pages.knowledge_base import show_knowledge_base
from app.pages.detection import show_detection
from app.pages.conversation import show_conversation
from app.pages.evaluation import show_evaluation
from app.pages.multi_algo_evaluation import show_multi_algo_evaluation

# 设置页面配置
st.set_page_config(
    page_title="DefectRAG - 产品表面缺陷检测系统",
    page_icon="🔍",
    layout="wide",
    initial_sidebar_state="expanded"
)

# 侧边栏标题
st.sidebar.title("DefectRAG 系统")
st.sidebar.markdown("基于 CN_CLIP + RAG + VLLM 的产品表面缺陷检测")

# 页面选择
pages = {
    "首页": show_home,
    "知识库管理": show_knowledge_base,
    "缺陷检测": show_detection,
    "对话模式": show_conversation,
    "系统测评": show_evaluation,
    "多算法测评": show_multi_algo_evaluation
}

# 选择页面
selection = st.sidebar.radio("选择功能", list(pages.keys()))

# 显示选择的页面
pages[selection]()

# 侧边栏底部信息
st.sidebar.markdown("---")
st.sidebar.info("DefectRAG v1.0 - 基于深度学习的产品表面缺陷检测系统") 